package com.ian.offer;

import com.ian.linked.TreeNode;

import java.util.HashMap;

/**
 * @author zhangyaowen
 * @date 2021/3/16 9:26 下午
 */

/**
 * 输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
 *
 *  
 *
 * 例如，给出
 *
 * 前序遍历 preorder = [3,9,20,15,7]
 * 中序遍历 inorder = [9,3,15,20,7]
 * 返回如下的二叉树：
 *
 *     3
 *    / \
 *   9  20
 *     /  \
 *    15   7
 *  
 *
 * 限制：
 *
 * 0 <= 节点个数 <= 5000
 *
 *  
 *
 * 注意：本题与主站 105 题重复：https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/
 *
 *
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/zhong-jian-er-cha-shu-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class BuildTree {

    private HashMap<Integer,Integer> map = new HashMap<>();

    public TreeNode buildTree(int[] preorder, int[] inorder) {

        int n = preorder.length;

        for (int i = 0; i < n; i++) {
            map.put(inorder[i],i);
        }




        return myBuild(preorder,inorder,0,n-1,0,n-1);
    }

    private TreeNode myBuild(int[] preorder, int[] inorder, int p_left, int p_right, int i_left,int i_right) {
            if (p_left>p_right){
                return null;
            }
        // 前序遍历中的第一个节点就是根节点
        int p_root = p_left;
        // 在中序遍历中定位根节点
        int i_root = map.get(preorder[p_root]);

        // 先把根节点建立出来
        TreeNode root = new TreeNode(preorder[p_root]);
        // 得到左子树中的节点数目
        int size_left_subtree = i_root - i_left;
        // 递归地构造左子树，并连接到根节点
        // 先序遍历中「从 左边界+1 开始的 size_left_subtree」个元素就对应了中序遍历中「从 左边界 开始到 根节点定位-1」的元素
        root.left = myBuild(preorder, inorder, p_left + 1, p_left + size_left_subtree,
                i_left, i_root - 1);
        // 递归地构造右子树，并连接到根节点
        // 先序遍历中「从 左边界+1+左子树节点数目 开始到 右边界」的元素就对应了中序遍历中「从 根节点定位+1 到 右边界」的元素
        root.right = myBuild(preorder, inorder, p_left + size_left_subtree + 1, p_right,
                i_root + 1, i_right);

        return root;
    }
}
